import polars as pl
import pandas as pd


def pl_test():
    df = pl.read_excel("polars_excel.xlsx", sheet_name="Sheet2")
    # print(df)
    print(df.to_dict())
    # print(df.head())
    # print(df['d'])
    pd.read_sql("polars_excel.sql", df)
    pl.sql("polars_excel.sql")


def pd_test():
    df = pd.read_excel("polars_excel.xlsx", sheet_name="Sheet2")
    df['avg'] = df[['d', 'e', 'f']].mean(axis=1)
    print(df)


if __name__ == '__main__':
    pd_test()
